list comprehensions solution in python